package com.simpleim.starter.handler;

import com.simpleim.common.task.FutureTaskScheduler;
import com.simpleim.im.common.bean.msg.ProtoMsg;
import com.simpleim.starter.server.ServerSession;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.handler.timeout.IdleStateEvent;
import io.netty.handler.timeout.IdleStateHandler;
import org.springframework.stereotype.Service;

import java.util.concurrent.TimeUnit;

@Service
@ChannelHandler.Sharable
public class HeartBeatServerHandler extends IdleStateHandler {
    private static final int READ_IDLE_GAP = 180;

    public HeartBeatServerHandler() {
        // READ_IDLE_GAP 时间内没有数据入栈，就判定连接假死
        // 第二个参数表示 出栈的时间间隔
        // 第三个参数 表示 出栈/入栈的时间间隔
        // 第四个参数，间隔的时间单位秒
        super(READ_IDLE_GAP, 0, 0, TimeUnit.SECONDS);
    }

    @Override
    protected void channelIdle(ChannelHandlerContext ctx, IdleStateEvent evt) throws Exception {
        System.out.println(READ_IDLE_GAP + "秒内未读到数据，关闭连接");
        ServerSession.closeSession(ctx);
    }

    @Override
    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
        if (!(msg instanceof ProtoMsg.Message)) {
            super.channelRead(ctx, msg);
            return;
        }

        ProtoMsg.Message message = (ProtoMsg.Message) msg;
        ProtoMsg.HeadType headType = message.getType();
        if (headType == ProtoMsg.HeadType.HEART_BEAT) {
            // 异步发送消息
            FutureTaskScheduler.add(() -> {
                // 连接是活动状态，才继续发送消息
                if (ctx.channel().isActive()) {
                    ctx.writeAndFlush(message);
                }
            });
        }

        super.channelRead(ctx, msg);

    }
}
